#!/usr/bin/env python
# -*- coding: utf-8 -*- 
############################################################################
#    Copyright (C) 2007-2008
#                  Ferraro Luciano (aka lux) <luciano.ferraro@gmail.com>
#                  Matto Scacco (aka checkm, scaccom) <matto.scacco@gmail.com>
#
#
#    This program is free software; you can redistribute it and/or modify 
#    it under the terms of the GNU General Public License as published by 
#    the Free Software Foundation; either version 2 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program; if not, write to the
#    Free Software Foundation, Inc.,
#    59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
############################################################################

import gtk
import gobject

class ChannelsList (gtk.TreeView):
    def __init__ (self):
        gtk.TreeView.__init__ (self)
        
        self.model = gtk.TreeStore(gobject.TYPE_STRING)
        self.model_map = {}
        
        renderer = gtk.CellRendererText()
        self.column = gtk.TreeViewColumn("Channels", renderer, text=0)
        self.selection = self.get_selection()

        self.append_column(self.column)        
        self.set_model(self.model)
        self.show_all()
        
    def add_tree (self, server_name):
        server_name_iter = self.model.insert_before(None, None)
        
        self.model.set_value(server_name_iter, 0, server_name)
        self.model_map[server_name] = [server_name_iter, {}]
        self.expand_all()
    
    def remove_tree (self, server_name):
        server_name_iter = self.model_map[server_name][0]
        
        self.model.remove(server_name_iter)
        del self.model_map[server_name]
    
    def add_subtree (self, server_name, channel_name):
        server_name_iter = self.model_map[server_name][0]
        channel_name_iter = self.model.insert_before(server_name_iter, None)
        
        self.model.set_value(channel_name_iter, 0, channel_name)
        self.model_map[server_name][1][channel_name] = channel_name_iter
        self.expand_all()
    
    def remove_subtree (self, server_name, channel_name):
        channel_name_iter = self.model_map[server_name][1][channel_name]
        
        self.model.remove(channel_name_iter)
        del self.model_map[server_name][1][channel_name]
